package com.dragon.datastructure.queue;

import com.dragon.datastructure.tree.heap.MaxHeap;

/**
 * 优先队列: 底层本质其实是 最大堆(完全二叉树 + 堆的性质)，用数组构建的最大堆
 * 与普通队列的区别就是: 普通队列是先入先出，而优先队列是根据元素的优先权来选择谁先出队列，而这种优先权是用户可以自己定义的
 *
 * @author dragon
 * @since 2020/1/1
 */
public class PriorityQueue<E extends Comparable<E>> {

    private MaxHeap<E> heap;

    public PriorityQueue() {
        heap = new MaxHeap<>();
    }

    public PriorityQueue(int capacity) {
        heap = new MaxHeap<>(capacity);
    }

    public PriorityQueue(E[] array) {
        heap = new MaxHeap<>(array);
    }

    public int size() {
        return heap.size();
    }

    public boolean isEmpty() {
        return heap.isEmpty();
    }

    /**
     * 入队
     *
     * @param e
     */
    public void enqueue(E e) {
        heap.add(e);
    }

    /**
     * 出队
     *
     * @return
     */
    public E dequeue() {
        return heap.extractMax();
    }


    public E peek() {
        return heap.findMax();
    }
}
